taking input in array in python